package com.zhaojk.audio.service;

import com.zhaojk.audio.entity.PromptTemplate;
import com.zhaojk.audio.mapper.PromptTemplateMapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.HashMap;
import java.util.Map;

@Service
public class PromptTemplateServiceImpl {
    @Autowired
    private PromptTemplateMapper promptTemplateMapper;
    @Autowired
    private VolcTextServiceImpl volcTextService;

    public String extendPrompt(String prompt, String translation){
        // template_id
        int id = 2;
        PromptTemplate promptTemplate = promptTemplateMapper.findById(id);
        //将提示词嵌入模板中
        Map variables = new HashMap();
        variables.put("prompt", prompt);
        variables.put("translation", translation);
        String templatedPrompt = TemplateProcessor.processTemplate(promptTemplate, variables);
        String extendedPrompt = volcTextService.chat(templatedPrompt);
        return extendedPrompt;
    }

    public String extendPrompt(String prompt){
        // template_id
        int id = 2;
        PromptTemplate promptTemplate = promptTemplateMapper.findById(id);
        //将提示词嵌入模板中
        Map variables = new HashMap();
        variables.put("prompt", prompt);
        variables.put("translation", prompt);
        String templatedPrompt = TemplateProcessor.processTemplate(promptTemplate, variables);
        String extendedPrompt = volcTextService.chat(templatedPrompt);
        return extendedPrompt;
    }
}
